package com.c2b.algorithm.leetcode.base;

import java.util.LinkedList;
import java.util.Queue;

/**
 * <a href="https://leetcode.cn/problems/sum-root-to-leaf-numbers/">求根节点到叶节点数字之和(Sum Root to Leaf Numbers)</a>
 * <p>给你一个二叉树的根节点 root ，树中每个节点都存放有一个 0 到 9 之间的数字。每条从根节点到叶节点的路径都代表一个数字：<p>
 * <li>例如，从根节点到叶节点的路径 1 -> 2 -> 3 表示数字 123 。</li>
 * 计算从根节点到叶节点生成的 所有数字之和 。
 * <p>叶节点 是指没有子节点的节点。</p>
 * <p>
 * <b>示例：</b>
 * <pre>
 * 示例 1：
 *      输入：root = [1,2,3]
 *      输出：25
 *      解释：
 *          从根到叶子节点路径 1->2 代表数字 12
 *          从根到叶子节点路径 1->3 代表数字 13
 *          因此，数字总和 = 12 + 13 = 25
 *
 * 示例 2：
 *      输入：root = [4,9,0,5,1]
 *      输出：1026
 *      解释：
 *          从根到叶子节点路径 4->9->5 代表数字 495
 *          从根到叶子节点路径 4->9->1 代表数字 491
 *          从根到叶子节点路径 4->0 代表数字 40
 *          因此，数字总和 = 495 + 491 + 40 = 1026
 * </pre>
 * </p>
 * <p>
 * <b>提示：</b>
 * <ul>
 *     <li>树中节点的数目在范围 [1, 1000] 内</li>
 *     <li>0 <= Node.val <= 9</li>
 *     <li>树的深度不超过 10</li>
 * </ul>
 * </p>
 *
 * @author c2b
 * @since 2023/5/11 16:10
 */
public class LC0129SumRootToLeafNumbers_M {

    static class Solution {
        public int sumNumbers(TreeNode root) {
            Queue<TreeNode> queue = new LinkedList<>();
            queue.offer(root);
            int sum = 0;
            while (!queue.isEmpty()) {
                TreeNode currNode = queue.poll();
                if (currNode.left == null && currNode.right == null) {
                    sum += currNode.val;
                } else {
                    if (currNode.left != null) {
                        currNode.left.val += currNode.val * 10;
                        queue.offer(currNode.left);
                    }
                    if (currNode.right != null) {
                        currNode.right.val += currNode.val * 10;
                        queue.offer(currNode.right);
                    }
                }
            }
            return sum;
        }

        public int sumNumbers2(TreeNode root) {
            return dfs(root, 0);
        }

        private int dfs(TreeNode node, int preSum) {
            if (node == null) {
                return 0;
            }
            preSum = preSum * 10 + node.val;
            if (node.left == null && node.right == null) {
                return preSum;
            }
            return dfs(node.left, preSum) + dfs(node.right, preSum);
        }
    }

    public static void main(String[] args) {
        //TreeNode root = new TreeNode(1, new TreeNode(2), new TreeNode(3));
        TreeNode root = new TreeNode(4);
        root.left = new TreeNode(9);
        root.right = new TreeNode(0);
        root.left.left = new TreeNode(5);
        root.left.right = new TreeNode(1);
        Solution solution = new Solution();
        System.out.println(solution.sumNumbers2(root));
    }
}
